package com.ruoyi.common.utils.barcode;

import com.ruoyi.common.enums.BarCodeEnum;
import com.ruoyi.common.utils.StringUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.ObjectUtils;
import org.krysalis.barcode4j.HumanReadablePlacement;
import org.krysalis.barcode4j.impl.code128.Code128Bean;
import org.krysalis.barcode4j.impl.code39.Code39Bean;
import org.krysalis.barcode4j.output.bitmap.BitmapCanvasProvider;
import org.krysalis.barcode4j.tools.UnitConv;
import java.awt.image.BufferedImage;
import java.io.*;

/**
 * 使用barcode4j生成条形码
 *
 * @author yinjc
 */
@Slf4j
public class BarcodeUtils {

    /**
     * 生成条形码文件
     *
     * @param msg  条形码的文本内容
     * @param path 生成条形码的文件路径
     * @return
     */
    public static void generateFile(String msg, String path, String barCodeType) throws IOException {
        File file = new File(path);
        OutputStream outputStream = null;
        try {
            outputStream = new FileOutputStream(file);
        } catch (FileNotFoundException e) {
            log.error("未找到文件!");
        }
        byte[] barCodeBytes = null;
        if (StringUtils.equals(barCodeType, BarCodeEnum.CODE128.getCodeType())) {
            barCodeBytes = generateBarCode128(msg, false);
        } else if (StringUtils.equals(barCodeType, BarCodeEnum.CODE39.getCodeType())) {
            barCodeBytes = generateBarCode39(msg);
        }
        try {
            if (null == barCodeBytes) {
                log.info("barCodeBytes为空！");
                return;
            }
            if(null == outputStream) {
                log.info("outputStream为空！");
                return;
            }
            outputStream.write(barCodeBytes, 0, barCodeBytes.length);
            outputStream.flush();
        } catch (Exception e) {
            throw new RuntimeException(e);
        } finally {
            try {
                if (null != outputStream) {
                    outputStream.close();
                }
            } catch (Exception e) {
                log.error("outputStream关闭异常！");
            }
        }
    }

    /**
     * 已生成code128条形码为例
     * @param msg           要生成的文本
     * @param hideText      隐藏可读文本
     */
    static byte[] generateBarCode128(String msg, boolean hideText) {
        if (StringUtils.isEmpty(msg)) {
            return null;
        }
        // 如果想要其他类型的条码(CODE 39, EAN-8...)直接获取相关对象Code39Bean...等等
        Code128Bean bean = new Code128Bean();
        // 分辨率：值越大条码越长，分辨率越高。
        final int dpi = 256;
        // 设置两侧是否加空白
        bean.doQuietZone(true);
        // 设置条码每一条的宽度
        // UnitConv 是barcode4j 提供的单位转换的实体类，用于毫米mm,像素px,英寸in,点pt之间的转换
        // 设置条形码高度和宽度
        bean.setBarHeight((double) ObjectUtils.defaultIfNull(10.00, 9.0D));

        bean.setModuleWidth(UnitConv.in2mm(3.0f / dpi));

        // 设置文本位置（包括是否显示）
        if (hideText) {
            bean.setMsgPosition(HumanReadablePlacement.HRP_NONE);
        }
        // 设置图片类型
        String format = "image/png";

        ByteArrayOutputStream ous = null;
        byte[] imageByte = null;

        try {
            ous = new ByteArrayOutputStream();
            BitmapCanvasProvider canvas = new BitmapCanvasProvider(ous, format, dpi,
                    BufferedImage.TYPE_BYTE_BINARY, false, 0);

            // 生产条形码
            bean.generateBarcode(canvas, msg);
            // 结束
            canvas.finish();
            imageByte = ous.toByteArray();
        } catch (IOException e) {
            log.error("IOException:" + e.getMessage());
        } finally {
            try {
                if (null != ous) {
                    ous.close();
                }
            } catch (Exception e) {
                log.error("ByteArrayOutputStream流关闭失败，失败原因为：{{}}", e.getMessage());
            }
        }
        return imageByte;
    }

    /**
     * 生成code39条形码
     *
     * @param msg
     */
    static byte[] generateBarCode39(String msg) {
        if (StringUtils.isEmpty(msg)) {
            return null;
        }
        Code39Bean bean = new Code39Bean();
        // 精细度
        final int dpi = 256;
        // module宽度
        final double moduleWidth = UnitConv.in2mm(1.0f / dpi);
        // 配置对象
        bean.setModuleWidth(moduleWidth);
        bean.setWideFactor(3);

        // 设置条码每一条的宽度
        // UnitConv 是barcode4j 提供的单位转换的实体类，用于毫米mm,像素px,英寸in,点pt之间的转换
        // 设置条形码高度和宽度
        bean.setBarHeight((double) ObjectUtils.defaultIfNull(10.00, 9.0D));

        // 设置两侧是否加空白
        bean.doQuietZone(true);

        String format = "image/png";
        ByteArrayOutputStream ous = null;
        byte[] imageByte;
        try {
            ous = new ByteArrayOutputStream();
            // 输出到流
            BitmapCanvasProvider canvas = new BitmapCanvasProvider(ous, format, dpi,
                    BufferedImage.TYPE_BYTE_BINARY, false, 0);
            // 生成条形码
            bean.generateBarcode(canvas, msg);
            // 结束绘制
            canvas.finish();
            imageByte = ous.toByteArray();
        } catch (IOException e) {
            throw new RuntimeException(e);
        } finally {
            try {
                if (null != ous) {
                    ous.close();
                }
            } catch (Exception e) {
                log.error("ByteArrayOutputStream流关闭失败，失败原因为：{{}}", e.getMessage());
            }
        }
        return imageByte;
    }

    public static void main(String[] args) throws IOException {
        String msg = "MOS123-CT1-A1";
        //生成条形码路径
        String path = "D:\\Users\\Desktop\\barcode.png";
        generateFile(msg, path, BarCodeEnum.CODE39.getCodeType());
    }

}
